Python Implementation for the orange track of "Feedback-stabilization of an arbitrary pure state of a two-level atom" paper by Wang and Wiseman