Clean up, added movement buttons on the screen