Added model test
updated dataset and model