adding controll to stages