updated TDD tests for syntax issues
added conditional to line 16 for the count function