Make the problem method a belongs_to association