Updated MotifTreeStateSearch to reflect changed in python