remove assertion that 3 columns are necessary

Since in the arguments you should now pass the number of columns