add python import code