Updete to HW 4