After class update