Add an initializer function for the train db