Added code to init the db