#ailab .lab-sim__badge {
  display: none;
}

#ailab .lab-sim__top {
  grid-template-columns: minmax(0, 1fr) auto;
}

#ailab .lab-sim__controls {
  grid-area: 1 / 2;
}

@media (max-width: 767px) {
  #ailab .lab-sim__top {
    grid-template-columns: minmax(0, 1fr);
  }

  #ailab .lab-sim__controls {
    grid-area: 2 / 1;
  }
}