Retirement Calculator in Python