Updated the shader so it looks more realistic