Added in class 3 assignment