Homework 2 python files