Adding UI changes that didnt commit correctly