Add docs & types