Add database models

For now this is just some test models, they will need to be changed later on.