Update
Add simpler test