Removed the transition matrix for now