Updated JS to match HTML