Suggest shortest path code

Look at bottom of the test class to see it