Refactor flexbox responsive styles