Fix UI + implement print (not finish)