A quick (and dirty!) fix for the error. #12
Merged Issue12 into main. closed #12