simple codes when learning python