Added code for nodes at a disctance of k from given node problem