Added code to get rook moves