Add the part of finding shortest path and made a straighforward test and it worked!