add parentheses and return statement to example code