improving example.py

Signed-off-by: Yaniv Bronhaim <ybronhei@redhat.com>