Cleanup
Updated todo