Har lagt till ett UI och ett interface till detta och ändrat lite i klasserna som anropar detta.