Add the setData function
Adding tests for setData