python run code only if main