Как сделать автоматическую разбивку элементов по колонкам?
 

Есть такой пример: https://plnkr.co/edit/vyAsUAJiEwcPD1ipbPad?p=preview

Можете подвигать границу окна, чтобы увидеть как элементы идут сверху вниз и разбиваются по колонкам (чем шире окно тем больше колонок). Так же виден баг, когда элементы обрезаются при переносе в другую колонку.

Нужно сделать то же самое, но без бага. Смотрю в сторону CSS Grid, но пока не понимаю как его приспособить. Есть идеи?