Use return statements instead of assign statements