some styling and directive tiddy up