Added solution to the last problem