Clean up JS