Lower batch size for more fancy demo
Save result screen as PNG
Implement multi processing