added some UI stuff