Refactor shader