working with python from amherst college comp sci class