Fix conversion of greyscale pix col
Move image mode conversion code to within main class
Add comments
Reenable demo training