clean code in python