Added small demo. Removed unused commented code