reviewd stein's code