fixed small bugs
Added small demo
updated readme