Make example into a test