Python function use global variable