finished UI part of it