all the code in python