Add models.py

Just some imports and Problem class. Feel free to rewrite.