Fix WASD movement + Fixed End Game UX (next level)