HW4 submission version