Fixes minor bugs and doc string typos