Small modification on architecture.
Add a function to create image montage.