So what would the new code look like? I'm not to proficient at JS.