Adds minor UI improvements