Treat return as a statement, not as a function.
Improve some docblocks.