JS added, further styling improvements