the flex-box works better now