Moved code to header file