Cleaned up Rook move check function

Cleaned up the function so that it only uses one loop
and making it (hopefully) more efficient.
Started implementing checkmate checking