Diagonal Checking

Missed out diagonal check....