Add docs.
Remove some useless import