Moving to flexbox for vertical components