Fix import , remove unused variables