added find_shortest_index and updated round_robin to have 2 arguements in driver