PYTHON: Added some modifications.  The python version now returns a list (dont know if we need array or not) and is similar in speed to julia.