How To Define A Math Function In Python