diagonal check works
Merge pull request #8 from sayshay/alex_andrew

diagonal check works