Some changes to your grunt task