fix: getting column names with names and reset index function

2 files changed, 3 insertions, 7 deletions

diff --git a/src/index.ts b/src/index.ts
index 99119a3..7c10f85 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -37,12 +37,8 @@ class Izuku {
       const izSampler = getSingleColumnDetails(this, column);
       return new Izuku(izSampler.rowd, izSampler.rowh);
     } else if (Array.isArray(column) || isArrayOfType(column, 'string')) {
-      if (isArrayOfType(column, 'number')) {
-        const izSampler = getMultipleColumnDetails(this, column);
-        return new Izuku(izSampler.rowd, izSampler.rowh);
-      } else {
-        throw new Error('Columns must be an array of integers or column names');
-      }
+      const izSampler = getMultipleColumnDetails(this, column);
+      return new Izuku(izSampler.rowd, izSampler.rowh);
     } else {
       throw new Error('Unexpected type of column');
     }
diff --git a/src/lib/frame.ts b/src/lib/frame.ts
index f81141d..aff6d31 100644
--- a/src/lib/frame.ts
+++ b/src/lib/frame.ts
@@ -26,7 +26,7 @@ export function header(
   this: Frame,
   header: Array<string>
 ): Array<string> | any {
-  if (!header.length) {
+  if (!header || header.length) {
     this.columns = generateHeader(this.rowdata);
   } else {
     this.columns = setHeader(this.rowdata, header);