Added special wrapper for flex box