remove unused import
refactor image generator