Add tests for Persistence module