Added missing 'use' statements