runtimewarning overflow encountered in square