Add a test to ensure data is being stored properly