prettify python code