Minor updates to test functions. Previous had absolute mean error rather than mean abs error - replaced with correlation check. Also added some graphical output to be comapred by eye