added proper model
added proper model