Add Memory clean up for Stage