Update grunt to watch JS files