Added tests for count and max