corrected train and test datasets
neural ntwork classfication added