Added solution for problem two