updated images and fixed bug