Diagonal Difference Hackerrank Python