Cleaned up fragment shader