add operator overloading to python version
clean up