Added a stage listener