added relations to existing models and tests

Fixes #15