Database connection added. Fixes #15