Add know return type for methods