minor revision

Making more classes implement Problem