diff --git a/helper.jl b/helper.jl
index fc7a71b..931fdfb 100644
--- a/helper.jl
+++ b/helper.jl
@@ -51,7 +51,7 @@ end
 
 "In-place c-orthogonalization via (modified) Gram-Schmidt. Only significant vectors are returned (c-normalized).
 The number of significant vectors to return are determined by the original singular values (compared to the threshold) "
-function gram_schmidt!(vecs, ws, threshold=1e-5; verbose=false)
+function gram_schmidt!(vecs, ws=ones(length(vecs[1])), threshold=1e-5; verbose=false)
     c_product(i, j) = sum(vecs[i] .* ws .* vecs[j])
     norm(i) = c_product(i, i)
     proj(i, j) = (c_product(i, j) / norm(i)) .* vecs[i] # component of vec[i] in vec[j]