Add test and update README with proper example