Indexing Into Numpy Array Python Code Checker