UI/UI bugs fixed