Fix css to better UI