Para alinear los cuadrados ahora en forma vertical, uno tras otro (sin espacios aún) hemos utilizado el flex-direction:column, posteriormente tendriamos que aplicar el align-items: center para que se alinearan centralmente. Puesto que al ser columna hemos afectado a los ejes de posicionamiento.