Improve functionnalities