Adding a simple test for flattening functionality.