stock prediction python code