Added option for testing database connection