working on UI of the POC