Updated testing setup and docs