Handling of return instructions added.