Added toy example simulation code