A mistake :(, fixed.
Some pep8 issues are fixed as well.