Added solution to assignment