clean python code