added a bit of ui