cleaned up the JS so that I can add more features, looking into local storage