add better UI graphics