Fix: add functions to header file