Python Code For Stock Trading