Fixed private functions

Updated private function