Partially fixed the DBHelper class