replace <main> classes with function
Easier to make changes on the grid and layout from a single place