Assignment 1, python