Add tests for Government class