added ui stuff