Ui is basically done