Add some changes on lecture 04