Overload update with class-specific code instead of doing it in the loop code