Better documentation, PEP8

Well, still not great, but should help.