Add model name attribute