Added solution to 1 more problem