Add status change handling