Add UI features and game pausing