Added some changes to the example