homework4_5 small update