Add diagonal check