UI is now working properly!