Fix README example and add more documentation