Rework flexbox mixins to use 'when' guard