https://toxigon.com/optimizing-python-code-with-numpy