Add test and change README