Started working on stage 4, added ids because those make more sense