Adding a new solution for assignment2 using nn lib