Responsive grid-like using flexbox.

This is a good example to show that even when the layout looks grid-like, as long as the Aspect Ratio of the boxes stays the same, it's still easier to make a 2D layout using flexbox instead of grid. I simplified the code a lot with this new solution.