Fix CS and add use statement