Code from assignment added to python file