Improve gruntfile