WIP HW1 - simulate function works ok