new problem set 1 code