Added stage model to keep track of where the user is on the level