White balance now works