clean up order of return type from util
completed set1 challenge 4