CSS updates

Initial use of flexbox