Added functionality to the sample