Update test and readme