added Problem class