Update __init__.pyx

prob calculation bug fixed