bug in exercise fixed