Very interesting! I've found a lot of numpy indexing confusing, so it's very nice to see some effort to clean it up!