Add castling, it works