shader fixes
Changes to the solution