Added solution to problem #4