Fixes #6 added database connection