Minor fixes
Updating sample