Cleaned up the design of the flexbox demo