HW5 update

Fix where issue