Added TODO item to improve the docs