added UI and did some fixes and balancing