Get occurrences for return keywords.

nit: can you use statement instead of stmt