fixed my hw #1