Add additional functionality to sample implementation.