works on toy example