add unit/integration tests