Update code per JN