Simple python codes for practice