Updating docs and moving tests outside package