Added stuff to cleanup section of stage3