Kane Williamson shows his class on return from injury, again