Image generation now working. Still full of bugs. Dont use in production!