Edited JS challenge