[Task] Add solution to assignment 1

Source codes for assignment 1