Updating to work with external model repo