hw2 update remove if statement