Update: improved scaling of saved classification images.