Doc update
Added more typing