removing the parameter: diffW, and add some annotations