**Just for comparing with the python code**