Max Aub, Teatro breve escrito en México. Con “Los muertos” y “El hombre del balcón”