Fixes #10 - make code PEP8 compatible.