code written while learning python