Updates code to the new version