Sum of three squares python. .