Updated to be pep-8 compliant.

@cranberyxl pls review