Add `@return` annotation