modified the models and added samples