Refactored code for fizzbuzz return