add sample images and code tweaks