Added example + small bugfix