added model class.. need to fill up the codes