PEP8 fixes
Add param and return notes