Use integer indexing only
Add missing import