Added missing import for LogReg classifier