Simple code editor in python