Fix docstring and coding style (E201 E202 E302)