fixed stage system and added restart button