Add shader wrapper