Fixed bug in challenge code