add Problem class