Added c4981 assignment 2