Added some more code for example