Make the JS mostly work. Make styling slightly prettier