Cleaned up imports,  removed showing image (doesn't work on RPi)