Address review comments and fix flake8
Address review comments and fix flake8

Signed-off-by: Alessandro Genova <a4b3c52a406e75d61cdb19d917d7654b6e73804b@kitware.com>

OpenChemistry/tomviz,OpenChemistry/tomviz,OpenChemistry/tomviz,OpenChemistry/tomviz