UI almost done
First UI done: need some improvement