Implement diagonal check in game