code when learning python