Improved the checkmate feature