updated problem 320 a bit...