update exercise5.1 with more comment