add training with just 1s instead of 1 and 0 option