Added return type info for methods