Assignment 4: 1 & 2 done