Shortest Transit

Correct errors and make it faster