Refactor flexbox mixin