Updated stage 4