Used autopep8 to normalize / cleanup code