added Problem class
relation
My best code