Update to problem code