Clean up our JS code