Use return rather than returning at the end