Change example to adopt it to solve the assignment task