Use model association

Fork me on GitHub