Changing example code to make it more clear