Update code for new data