JS Homework - small fixes