Finally have ui working