Use flexbox for grid

Fixes #71