Changed pixel shader to use inline if.