Image gen working, sorta - tweaky.