HW4 replace with content from course