Add tests for completion