Added solution for assignment 3