Support for different image output types

Support for different image output types