Added return statement to extension method