Add data simulation code