Add tests for count_if
Add tests for count
Added drop_while