fixing minor UI bugs