Updates to Ui and stuff