(refactor) Use flex-box.