Clean up TODO tag and fix cs-setup import function