Modifying README example and Test