Refactored to include displayModelName