Added support for return keyword.