Edited example to a working example