adding test set
using train split manually