Fixed minor error
Minor UI updates