Fixed end game and checkmate