cleanup and adding enter on stage won to go to stage choose level