When a person dies, do they immediately awake to either heaven or hell? Or is their soul still in the grave?

awaiting judgment from God; then they'll know their final demise. I think I know the answer to this but I'm not really sure?