Added support for different parameters, and added images for rgb