Refactor tests, remove docs