Assignment 3 : part 1 done