All my work in python