Update code in class