Add public to necessary func