Refactor grunt setup with just in time loader