Added wrapper for shortest path algorithm in environment.py