python square a number