Add update code & enable updater