Docs polish
Clean up
PEP8 polish