code during learning python