Cleaned up fragment shader a bit