ENH: improve example + allow for adding single vectors