I fixed the bug where the cube stopped moving for some reason and I also added walls to make sure it's not buggy