Update code to new version