updated css to use flexbox