Add flex styling