Improve test integration and update docs