Make the code more python.