Fixed not PEP8 codes
Included some docstrings