Add expect.element function