Add space functionality and remove need to press space at end of game