Added code for assigment 2