Updated code
Added demo