refactoring w/ flexbox