corrected mistake in homework 1