Restructured shader code