[Prototype] UI part is almost done