.container {  
    display: flex;  
    flex-direction: column;  
  }