Support for FSM visualization. Converted distance to Utility
Updated graph image output
Updated graph generation to make it neat and easier to understand