Bug fix, columns can be define on any element now

Calculation was made only on DIV elements before.