Fixed the function issue!