Added proper relationship response to models