Addressed comments for second assignment.