Touching only diaganols on update.

This probably won't make a huge difference when K is small, but it's better style.