Refactored shader