Added missing 'use' statements to controller