Add problem 6 to unit 3