Add data logging