Update based on feedback from Stein