Add docs & Fix typing errors