doing math with python