Extra Credit HW assignment