Alter mixin to handle list as well as number of cols