Angular directive that creates balanced columns