Adding update support