HW week 1, added class files for week 2