Added python main snippet