Fixed test error, more updates to docs