Python Math and Augmented Assignment