Remove some of the dialyzer warnings and add some doc arg types