Add shader wrapper. Not finished yet...