Re-added Walljump + modified physics for the square