fix css to use flex box