Added persistence with in memory db