TODO : finalize UI