Applying changes on test_stateful as suggested by @nicoddemus

Tks for the tip! ;)
Updating documentation about stateful entities.