Added function for staging