Added my homework for defining classes