More fixes to enemy movement and a function to go back to main menu at any time which will reset the level