modified my training output