problem 3b fixes