added lr avg step