Added the code for Assinment 2