Updated game control flow