Building a Simple Autocorrelation Model in Python