Improved demo files for better understanding

Improved demo files for better understanding