Update the test and README