It would be good to point to the paper where you got the implementation from, and maybe to clarify how the implementation relates to what they describe there.