Doc improvements + PEP8