Cleaning up flexbox