Add mask image output format. Need to do it in a polymorphic way