Added some styling and JS