Use function wrapper and only expose classes