Small refactor.
Add join model.