.button {
  display: none;
}

@media only screen and (max-width: 768px) {
  .button {
    display: flex;
  }
}