Started messing with JS, still not working fully