Add some code for beginning db interaction