update assignment04, 05