Add checks on update script