Add test entity data parameter
Improve readme