Just trying some basic import function in python