For some reason, when I tried to use the numpy solution, it passed the initial test, but failed the actual test. I did it the more drawn out method (for loop) to pass and then looked at the other solutions. I saw "import math" suggested, and so tried it with that approach, which worked. I wonder why numpy method didn't work for me...