Add basic model functionality