Add DB connection

Added some DB functionality for testing purpose.