codes in python