Improve multi task support.

I could not get the master version to work on my grunt 0.4.0-rc6 version.

Have made some small changes that made it work nicely for me.