Adding way to add transition rules and unit tests