update the code a little to make it clear for students to use
update