Minor improvements in fizz_buzz