removed images and touched up logic a bit