Add in relationships and the next step in staging