fixed class name
Changed image writing directory
Changed image writing directory