College python program #1