Updated library to use new filter function.
Unit manager now stores the node that finished the task.