Ui updates

@jasonthomas: LMK if i need to change stuff