Add tests for persisting new data