Cool study @itsmuriuki - have you tried running it in one of our Jupyterlab environments?