modificado el main con lo agregado