homework 4 & 5 added. Fix HW4