new problem set 3 code