Update __init__.py

Removing hardcoded '3' and replacing it with len()