Added a quick viewer for some test and train images.