updated images and cleaned up codes