First draft in python