Cleaning up styling of JS