handle end of level & end of game way better