update for HW3-3 example