make indexing of x_correct adn x_time 1d; it was breaking for my version of numpy