update code for data example