refactored code
clean up image folder