Added option to keep selector focused on buttons after buying an upgrade