HW5 addition