Got rid of probability from state and modified str for better representation for dict