Added code to interact with db