The QENS notebook makes up almost half of the total execution time (~5 out of 12 min). Would be nice if we can reduce the runtime. This might be related to #101 ? If this part takes a long time to run, we might be removing it anyway.