For second assignment