fixed small bugs with restart and new game