added 'win' screen, game can properly be restarted