Added a task category for improvements to TODO.md