Create choices.py

working with the import math function