Implemented restart button
Made restart button visible on game end