Add assoc and update_in

Fixes #6.