Restart button fix

Added button to game screen, and both lose and win screens