Add data model classes

Closes #9