Made test_math more complex