Expertly fixed my code