Added a TODO.
Conditionally import the collections module.