Flexbox mixin refactor