Update happy_numbers.py