Modify method return statements