Updating docs & new tests