Add flex-grow and flex-shrink to the flexbox implementation