fixed bugs adn UI