Update the grunt file to be more flexible

Fixing #2