small code for learning python