function to reshape the data