Updated the example, to make it a bit better