integrate new e_step code into the python