test code in a function
fixed case when applying preproc on test dataset
Added docstring and made new preproc classes more general