Applying Math with Python