Added main back to DBConnection class