Updated nextStage function

- Uses listiterator instead of an index increment