problem 5 seems to be working?