The source code is borrowed from Jon Barron's public academic website: https://jonbarron.info/. Feel free to clone this code for your own personal use.