Add shader code