Use return instead of print