Added code for assignment - 1 which was missed out