Introduced some changes to the example