Upgrade single-prompt
Clean up how UI is used
Add main game loop