Fix play loop, check and checkmate methods