Add association to models
Merge pull request #49 from shelling/master

Add association to models