FIX NameError: global name 'math' is not defined