Succesfully implented 1st problem