Reformat data, add test