Improve CSS with mixins and flexbox