Start adding model-specific functionality