Further changes for preparing Train Set for WEKA

Signed-off-by: Tarjei400 <adrian.jutrowski@gmail.com>