Add some db code