[Fix]: adding public headers