Add tests for optional