improving test and readme