Improve comments, remove import math

Improve comments
Remove import math. Aren't using anything in math, so just remove it.