Adds stage size parameter