updated output format
separate training and testing