CSS Masonry Layout with Grid and JS