Added in the NUbugger code to output the classified images