Updated Return type