Adding updated UI