Change stage to use spritebatch