Ui fixes sprint2