Update with Petr's addition to simulate data