Added solution to problem three