Added import math...

Added a small optimization that should make the tool run smoother. Thanks to brunoais for finding this optimization...