Add ability to connect and query database