Refactor Number Problem.py