Added import math

Added comments on what needs to be done for exercise 2