remove unused import

math.log is not used anywhere.

Also added some whitespace by suggestion of pep8