added buttons to select stage