fixed bug when calling the new Ui class, needs the new class arguments