bugs fixed for level 1 training and test