Add tests and changes to update updated_at