Additional work on flexbox layout