python learn: first task