Add version update logic