1st problem correction