Solved bug in moment 6 lab 2