Minor Update

Allow user to specify which db to use.