Added return expression