Fixing shader code -_-