Don't use `count()` unless you need it

Here it does not make a huge difference but it's a bad practice.

PS: Please, please follow PEP-8.