Pep8 fix.
Updated for version 0.2.1.