added labels back into stages