Added .py files for hw2