added some nice ui features