TODO: add code example usage in docs