Cleanup tests.

No need using `div.class`. We can simply use `.class`
Add some extra fields to Problem model :smile:

Fields added are:
author, createdAt, updatedAt, inputs and outputs