added return to base and victory screens