continue working on Problem class