Added some documentation, PEP8 changes