enhance class update