Fixed up some style problems in the JS