Added some tests for GroupBy