Add import for DatabaseHelper