Add a little functionnality