.code-container {
    position: relative;
    background-color: #1a1a1a;
    color: #f5f5f5;
    border-radius: 8px;
    border: 1px solid #333;
    margin: 1rem 0;
    font-family: monospace;
    overflow: hidden;
  }

  .code-header {
    display: flex;
    justify-content: space-between;
    align-items: center;
    background: #4a4; /* blue header */
    color: #fff;
    padding: 6px 10px;
    font-size: 0.9rem;
    font-weight: bold;
  }

  .collapse-btn {
    background: none;
    border: none;
    color: #fff;
    cursor: pointer;
    font-size: 0.9rem;
    padding: 2px 6px;
    border-radius: 4px;
    transition: background 0.2s;
  }

  .collapse-btn:hover {
    background: rgba(255, 255, 255, 0.1);
  }

  .code-content {
    padding: 1rem;
    display: block;
    white-space: pre;
    overflow-x: auto;
    color: #fff8dc;
  }

  .collapsed .code-content {
    display: none;
  }

  .code-content {
    padding: 1rem;
    display: block;
    white-space: pre;
    overflow-x: auto;
    background-color: #1a1a1a;
    color: #f5f5f5;
    font-family: monospace;
    border-top: 1px solid #333;
  }

  /* Syntax colors */
  .kw { color: #82aaff; font-weight: bold; }   /* keywords */
  .fn { color: #c792ea; }                      /* functions */
  .str { color: #c3e88d; }                     /* strings */
  .num { color: #f78c6c; }                     /* numbers */
  .com { color: #c3e88d; font-style: italic; } /* comments */
  .class{color:#f78c6c; font-weight: bold; }
  .ret{color:#d6b83d}