Started work on toy model