diagonal difference python solution