Added missing includes.
Fixed docs.