Fix minor issues in styling