Move tests into module
Update documentation