Add debug statements while ironing out correct values for Ema and Stochastic