Changed the interface of Sean's example.

Signed-off-by: Marcin Zalewski <marcin.zalewski@gmail.com>