Improve tests in readme