Quick fix on week 1 js