modify a bit
update both test code and readme