Added stage control