Part 2 Working?

fix out-of-bounds error