I don't think that Punishment in the grave is from the Quran. I think its from the hadith(aleged sayings of Muhamed that were first recorded by a man called Bukhari 200 years after death of Muhamed). But I could be wrong.
Thanks
Thanks