Turns out when you change your interface you have to change your example codes too... (#8)
Implementing test_correlation_functions and fixing up some of the other tests (#8)