Add database initialisation and MockDB