Update problem 2.1.c

Removing some unnecessary lines of code.