Added collision box for stage 3 + 4