Add initial code for assignment 2

Add initial code for assignment 2