Add css and js for a better UI