Clean up
TODO update