make the example a bit more versatile