add demo images
update visualize function