Assignment #1: Python Solution