clean up code
clean up code
added StreamingContext.remember