improving tests and readme