Add model functionality