Adds ability to pass in which relationships should be returned