Updated to be pep-8 compliant.

@cranberryxl please review