Image generation is hopefully working