Simple Python script to do backtesting on stock data using different time windows