Add restart button
Add game ending on number of steps exceed