Modify initializer for new entity relationship