Quick tidyup
Update TODO