Add: answer for assignment 2
Add: code for assignment 2