improved the JS and added class to find things easier