added import

Learning how to import to Python in Python