add select screen logic