Cleanup and add todos.