small cleaning
changed test set