Some minor UI changes made #12