Improve example a bit