Added a win/lose message and rest button